package com.example.demo.leetcode.top100;

import java.util.Arrays;

/**
 * ******************************************************
 *
 * @author liugh9
 * @version 1.0
 * @classname _89分割等和子集
 * @description
 * @date 2023/07/01 9:43
 * <p>
 * ******************************************************
 */
public class _89分割等和子集 {

    /**
     * 给你一个 只包含正整数 的 非空 数组 nums 。请你判断是否可以将这个数组分割成两个子集，使得两个子集的元素和相等。
     * <p>
     * <p>
     * <p>
     * 示例 1：
     * <p>
     * 输入：nums = [1,5,11,5]
     * 输出：true
     * 解释：数组可以分割成 [1, 5, 5] 和 [11] 。
     * 示例 2：
     * <p>
     * 输入：nums = [1,2,3,5]
     * 输出：false
     * 解释：数组不能分割成两个元素和相等的子集。
     *
     * @param nums
     * @return
     */
    public boolean canPartition(int[] nums) {
        int sum = Arrays.stream(nums).sum();
        if (sum % 2 == 1) {
            return false;
        }
        int target = sum / 2;
        int length = nums.length;

        boolean[] dp = new boolean[target + 1];
        dp[0] = true;

        for (int i = 1; i < length; i++) {
            for (int j = target; j >= 0; j--) {
                if (j - nums[i] >= 0 && !dp[i]) {
                    dp[j] = dp[j - nums[i]];
                }
            }
        }

        return dp[target];
    }


    public boolean canPartition1(int[] nums) {
        int sum = Arrays.stream(nums).sum();
        if (sum % 2 == 1) {
            return false;
        }
        int target = sum / 2;
        int length = nums.length;

        // 第i个物品容量是j的时候是否可以装满
        boolean[][] dp = new boolean[length + 1][target + 1];
        // 容量为0代表可以装满
        for (int i = 0; i <= length; i++) {
            dp[i][0] = true;
        }

        for (int i = 1; i <= length; i++) {
            for (int j = 1; j <= target; j++) {
                if (j - nums[i - 1] < 0) {
                    dp[i][j] = dp[i - 1][j];
                } else {
                    dp[i][j] = dp[i - 1][j] || dp[i - 1][j - nums[i - 1]];
                }
            }
        }

        return dp[length][target];
    }
}
